package org.usmile.algorithms.leetcode.middle;

import java.util.ArrayList;
import java.util.List;

/**
 * 93. 复原 IP 地址
 *
 * 有效 IP 地址 正好由四个整数（每个整数位于 0 到 255 之间组成，且不能含有前导 0），整数之间用 '.' 分隔。
 * 例如："0.1.2.201" 和 "192.168.1.1" 是 有效 IP 地址，但是 "0.011.255.245"、"192.168.1.312" 和 "192.168@1.1" 是 无效 IP 地址。
 * 给定一个只包含数字的字符串 s ，用以表示一个 IP 地址，返回所有可能的有效 IP 地址，这些地址可以通过在 s 中插入 '.' 来形成。你 不能 重新排序或删除 s 中的任何数字。你可以按 任何 顺序返回答案。
 *
 * 示例 1：
 * 输入：s = "25525511135"
 * 输出：["255.255.11.135","255.255.111.35"]
 *
 * 示例 2：
 * 输入：s = "0000"
 * 输出：["0.0.0.0"]
 *
 * 示例 3：
 * 输入：s = "101023"
 * 输出：["1.0.10.23","1.0.102.3","10.1.0.23","10.10.2.3","101.0.2.3"]
 *
 * 提示：
 * 1 <= s.length <= 20
 * s 仅由数字组成
 */
public class _0093 {
}

class _0093_Solution {
    private List<String> res;

    public List<String> restoreIpAddresses(String s) {
        res = new ArrayList<>();

        restoreIp(s, 0, "", 0);

        return res;
    }

    private void restoreIp(String s, int index, String ip, int count) {
        if (count == 4) {
            if (index == s.length()) {
                res.add(ip);
            }

            return;
        }
        for (int segmentLen = 1; segmentLen < 4; segmentLen++) {
            if (index + segmentLen > s.length()) {
                break;
            }
            String segment = s.substring(index, index + segmentLen);
            if (!isValidIpSegment(segment)) {
                continue;
            }
            String suffix = count == 3 ? "" : ".";
            restoreIp(s, index + segmentLen, ip + segment + suffix, count + 1);
        }
    }

    private boolean isValidIpSegment(String segment) {
        // 长度不能大于 3
        int len = segment.length();
        if (len > 3) {
            return false;
        }

        // 1. ip 段如果是以 0 开始的话，那么这个 ip 段只能是 0
        // 2. ip 段需要小于等于 255
        return (segment.charAt(0) == '0') ? (len == 1) : (Integer.parseInt(segment) <= 255);
    }
}